# -*- coding: utf-8 -*-
"""
@Time    : 2024/7/9 19:42 
@Author  : ZhangShenao 
@File    : 1.生成情人节玫瑰宣传语.py 
@Desc    : 使用LangChain,生成情人节玫瑰宣传语
"""
import os

import dotenv
from langchain_openai import OpenAI

dotenv.load_dotenv()

llm = OpenAI(
    model_name='gpt-3.5-turbo-instruct',
    openai_api_base=os.getenv('OPENAI_API_BASE')
)

output = llm.invoke('请帮我写一段中文的情人节红玫瑰宣传语，要求字数在30字以内。')
print(output)
